Python Challenge: Candy Distribution Problem